import torch.nn as nn
import torch

class Perceptron(nn.Module):

    def __init__(self):
        super(Perceptron, self).__init__()
        self.hidden = nn.Linear(2, 1)

    def forward(self, x):
        return torch.sigmoid(self.hidden(x))